"""
公共方法模块，习惯命名为utils
"""
from time import sleep
from selenium import webdriver


class DriverUtil:
    """浏览器对象管理类"""

    driver = None

    def get_driver(self):
        # 为了防止在同一次测试过程中，调用获取浏览器对象方法时，
        # 都会创建一个新的浏览器对象，因此要先判断对象是否存在，不存在时在创建
        if self.driver is None:
            self.driver = webdriver.Chrome()
            self.driver.maximize_window()
            self.driver.implicitly_wait(10)
            self.driver.get("http://127.0.0.1")
        return self.driver

    def quit_driver(self):
        sleep(2)
        self.driver.quit()
        self.driver = None


if __name__ == '__main__':
    my_driver = DriverUtil()
    my_driver.get_driver()
    my_driver.quit_driver()
